#pragma once
#include <pthread.h>
#include <string>
#include <functional>
#include <cstdint>
namespace ThreadModule
{
    static uint32_t number = 1;
    class Thread
    {
        using func_t = std::function<void()>;

    private:
        // 启用分离
        void EnableDetach()
        {
            // 将_isdetach设置为true
            _isdetach = true;
        }
        // 启用运行
        void EnableRunning()
        {
            // 将_isrunning设置为true
            _isrunning = true;
        }
        /**
         * @brief 线程执行例程函数
         *
         * 该函数是线程启动后执行的例程函数，负责线程的初始化、设置名称以及执行用户定义的函数。
         *
         * @param args 传入的参数，实际类型为 Thread*，指向当前线程对象的指针
         * @return 返回值为 nullptr，表示线程执行结束
         */
        static void *Routine(void *args)
        {
            // 将传入的参数转换为 Thread* 类型
            Thread *self = static_cast<Thread *>(args);
            
            // 调用线程对象的 EnableRunning 方法，设置线程为运行状态
            self->EnableRunning();
            
            // 检查线程是否为分离状态，如果是则调用 Detach 方法进行分离
            if (self->_isdetach)
                self->Detach();
            
            // 使用 pthread_setname_np 函数设置线程名称
            pthread_setname_np(self->_tid, self->_name.c_str());
            
            // 调用线程对象的用户定义函数
            self->_func();
            
            // 返回 nullptr 表示线程执行结束
            return nullptr;
        }

    public:
        /**
         * @brief 构造函数，用于初始化线程对象。
         * 
         * @param func 线程执行的函数指针。
         * 
         * 构造函数初始化线程对象，并设置线程的初始状态。
         * 初始状态包括线程ID为0，未分离状态，未运行状态，
         * 返回值为空指针，以及将函数指针赋值给成员变量。
         * 同时，为线程命名，格式为"thread-"后跟一个自增的编号。
         */
        Thread(func_t func)
            : _tid(0),                  // 线程ID初始化为0
              _isdetach(false),         // 线程初始状态为未分离
              _isrunning(false),        // 线程初始状态为未运行
              _res(nullptr),            // 线程返回值初始化为空指针
              _func(func)               // 将传入的函数指针赋值给成员变量
        {
            _name = "thread-" + std::to_string(number); // 为线程命名，格式为"thread-"后跟自增编号
        }
        /**
         * @brief 分离当前线程
         * 
         * 该函数用于分离当前线程，使其在完成后自动释放资源。
         * 如果线程已经处于分离状态，则函数直接返回。
         * 如果线程正在运行，则调用pthread_detach函数进行分离。
         * 最后，调用EnableDetach函数来更新线程状态。
         */
        void Detach()
        {
            // 检查线程是否已经处于分离状态
            if (_isdetach)
                return;
            
            // 检查线程是否正在运行
            if (_isrunning)
                // 调用pthread_detach函数分离线程
                pthread_detach(_tid);
            
            // 更新线程状态为分离状态
            EnableDetach();
        }
        /**
         * @brief 启动线程。
         *
         * 该函数用于启动一个新的线程来执行指定的任务。如果线程已经处于运行状态，
         * 则返回 false，否则尝试创建一个新的线程并返回 true。
         *
         * @return 如果线程成功启动，则返回 true；否则返回 false。
         */
        bool Start()
        {
            // 检查线程是否已经在运行
            if (_isrunning)
                return false;
            
            // 创建一个新的线程，执行 Routine 函数，并将 this 传递给它
            int n = pthread_create(&_tid, nullptr, Routine, this);
            
            // 检查线程创建是否成功
            if (n != 0)
                return false;
            else
                return true;
        }
        /**
         * @brief 停止正在运行的线程。
         *
         * 该函数用于停止一个正在运行的线程。首先检查线程是否正在运行，如果是，则尝试取消该线程。
         * 如果取消线程成功，则返回 true；否则返回 false。
         *
         * @return 如果线程成功被取消，则返回 true；否则返回 false。
         */
        bool Stop()
        {
            if (_isrunning)
            {
                int n = pthread_cancel(_tid);
                if (n != 0)
                    return false;
                else
                    return true;
            }
            return false;
        }
        /**
         * @brief 等待线程结束
         * 
         * 该函数用于等待由 _tid 标识的线程结束。如果线程已经被分离（detached），则该函数不执行任何操作。
         * 
         * @note 
         * - 如果线程已经被分离，则该函数立即返回。
         * - 使用 pthread_join 函数等待线程结束，并将返回值存储在 _res 中。
         * - (void)n 用于忽略 pthread_join 的返回值，避免编译器警告。
         */
        void Join()
        {
            if (_isdetach)
                return;
            int n = pthread_join(_tid, &_res);
            (void)n;
        }
        pthread_t Id()
        {
            return _tid;
        }
        std::string Name()
        {
            return _name;
        }
        ~Thread()
        {
        }

    private:
        pthread_t _tid;
        std::string _name;
        bool _isdetach;
        bool _isrunning;
        void *_res;
        func_t _func;
    };
} // ThreadModule